#!/bin/bash
cd ..

MODEL='densenet_cifar'
EPOCHS=200
LR=0.1
BATCH_SIZE=128

export CUDA_VISIBLE_DEVICES=0,1,2,3
python -m torch.distributed.launch --nproc_per_node 4 --master_port 29503 train.py \
--choose_model ${MODEL} \
--epochs ${EPOCHS} \
--lr ${LR} \
--batch_size ${BATCH_SIZE}